import os

# 设置Tensorflow的运行日志等级为2--Warning
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import tensorflow as tf

if __name__ == '__main__':
    # 传入的参数作为方阵，对角线为1
    # 注意：单位矩阵只需要传入行，并且行的值类型只能是 数值，不能是 列表。
    e1 = tf.eye(2)
    print(e1)

    e2 = tf.eye(3)
    print(e2)
